# torchrun --nproc_per_node=2 generate.py  \
# --dit_fsdp \
# --t5_fsdp \
# --ulysses_size 2 \
# --offload_model False --task ti2v-5B --size 1280*704  --ckpt_dir /home/weights/Wan-AI/Wan2.2-TI2V-5B-BF16 --prompt "Two anthropomorphic cats in comfy boxing gear and bright gloves fight intensely on a spotlighted stage"  --frame_num 48

# torchrun --nproc_per_node=6 generate.py --task ti2v-5B \
# --dit_fsdp \
# --t5_fsdp \
# --size 1280*704 \
# --ulysses_size 6 \
# --offload_model False \
# --ckpt_dir /home/weights/Wan-AI/Wan2.2-TI2V-5B-BF16 \
# --prompt "Two anthropomorphic cats in comfy boxing gear and bright gloves fight intensely on a spotlighted stage" \
# --frame_num 48


torchrun --nproc_per_node=6 generate.py --task ti2v-5B \
--size 1280*704 \
--ulysses_size 6 \
--offload_model true \
--ckpt_dir /data/nlp/Wan2.2-TI2V-5B-BF16 \
--prompt "Two anthropomorphic cats in comfy boxing gear and bright gloves fight intensely on a spotlighted stage" \
--frame_num 121 --base_seed 24 --convert_model_dtype